import tensorflow as tf 

# tf.function可以把python函数通过autograph转换成tensorflow的图结构 在GPU上的运行速度快几十倍 (也可以通过tf.autograph.to_code转换成tf的代码 但是可读性不强)

@tf.function
def sum(x):
    total = tf.constant(0)  # 虽然定义的常量是tf的 但是这个函数本身是python的语法格式 是python的函数 
    increment = tf.constant(1)
    for _ in range(x):
        total += increment
        increment += 1
    return total


# 定义变量时要定义在函数外面
var = tf.Variable(0) 
@tf.function
def add_10():
    return var.assign_add(10)

if __name__ == "__main__":
    print(sum(100))
    print(add_10())
    